from utils import WeChatOfficialAccount, ZaoBaoAPI
import os

# 使用示例
if __name__ == "__main__":
    res_data = ZaoBaoAPI().get_daily_news()
    data = res_data.get("data")  # 修复变量名错误
    # 获取当前文件所在目录的绝对路径
    current_dir = os.path.dirname(os.path.abspath(__file__))
    # 组合成模板路径（假设模板目录在项目根目录的 templates 文件夹）
    template_path = os.path.join(current_dir,  "templates")

    wechat = WeChatOfficialAccount(template_path)

    # 上传图片示例
    media_data = wechat.upload_image(data["head_image"], is_online=True)
    data["head_image"] = media_data["url"]

    # 创建草稿示例
    article = {
        "title": f"【{data['date']}】每日60秒读懂世界",
        "author": "小录",
        "digest": data["weiyu"],
        "need_open_comment": 1,
        "content": wechat.generate_news_html(data),
        "thumb_media_id": media_data.get("media_id")
    }

    draft_id = wechat.create_draft([article])
    wechat.public_article(draft_id)
